package com.zzyl.nursing.job;

import cn.hutool.core.text.CharSequenceUtil;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.json.JSONObject;
import cn.hutool.json.JSONUtil;
import com.zzyl.framework.config.properties.HuaWeiIotConfigProperties;
import com.zzyl.nursing.dto.iot.IotMsgNotifyData;
import com.zzyl.nursing.service.IDeviceDataService;
import lombok.extern.slf4j.Slf4j;
import org.apache.qpid.jms.*;
import org.apache.qpid.jms.message.JmsInboundMessageDispatch;
import org.apache.qpid.jms.transports.TransportOptions;
import org.apache.qpid.jms.transports.TransportSupport;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import org.springframework.stereotype.Component;

import javax.jms.*;
import java.net.InetAddress;
import java.net.URI;
import java.net.UnknownHostException;
import java.text.MessageFormat;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
import java.util.stream.Collectors;

/**
 * AMQP客户端类，用于连接华为云IoT平台并接收设备消息
 * 实现ApplicationRunner接口，在Spring Boot应用启动后自动运行
 *
 * @author itcast
 */
@Slf4j // Lombok注解，自动生成日志对象
@Component // Spring组件注解，将该类注册为Spring Bean
public class AmqpClient implements ApplicationRunner {

    // 自动注入华为云IoT配置属性
    @Autowired
    private HuaWeiIotConfigProperties huaWeiIotConfigProperties;

    // 业务处理异步线程池，线程池参数可以根据您的业务特点调整，或者您也可以用其他异步方式处理接收到的消息。
    @Autowired
    private ThreadPoolTaskExecutor threadPoolTaskExecutor;

    // 控制台服务端订阅中消费组状态页客户端ID一栏将显示clientId参数。
    // 建议使用机器UUID、MAC地址、IP等唯一标识等作为clientId。便于您区分识别不同的客户端。
    private static String clientId;

    // 静态代码块，在类加载时执行，用于初始化clientId
    static {
        try {
            // 使用本机IP地址作为clientId
            clientId = InetAddress.getLocalHost().getHostAddress();
        } catch (UnknownHostException e) {
            // IP地址获取失败时打印异常堆栈
            e.printStackTrace();
        }
    }

    /**
     * Spring Boot应用启动后执行的方法
     *
     * @param args 应用启动参数
     * @throws Exception 启动过程中可能抛出的异常
     */
    @Override
    public void run(ApplicationArguments args) throws Exception {
        // 启动AMQP客户端连接
        start();
    }

    /**
     * 启动AMQP客户端连接，创建指定数量的连接并监听消息
     *
     * @throws Exception 连接过程中可能抛出的异常
     */
    public void start() throws Exception {
        // 参数说明，请参见AMQP客户端接入说明文档。
        // 根据配置的连接数创建多个AMQP连接
        for (int i = 0; i < huaWeiIotConfigProperties.getConnectionCount(); i++) {
            // 创建amqp连接
            Connection connection = getConnection();

            // 加入监听者，监听连接状态变化
            ((JmsConnection) connection).addConnectionListener(myJmsConnectionListener);
            // 创建会话，使用自动确认模式
            // Session.CLIENT_ACKNOWLEDGE: 收到消息后，需要手动调用message.acknowledge()。
            // Session.AUTO_ACKNOWLEDGE: SDK自动ACK（推荐）。
            Session session = connection.createSession(false, Session.AUTO_ACKNOWLEDGE);
            // 启动连接
            connection.start();

            // 创建Receiver连接，监听指定队列的消息
            MessageConsumer consumer = newConsumer(session, connection, huaWeiIotConfigProperties.getQueueName());
            // 设置消息监听器处理接收到的消息
            consumer.setMessageListener(messageListener);
        }

        // 记录日志，表示AMQP客户端已成功启动
        log.info("amqp is started successfully, and will exit after server shutdown");
    }

    /**
     * 创建amqp连接
     *
     * @return amqp连接
     */
    private Connection getConnection() throws Exception {
        // 生成连接URL
        String connectionUrl = generateConnectUrl();
        // 创建JMS连接工厂
        JmsConnectionFactory cf = new JmsConnectionFactory(connectionUrl);
        // 创建传输选项并设置信任所有证书
        // 信任服务端
        TransportOptions to = new TransportOptions();
        to.setTrustAll(true);
        // 设置SSL上下文
        cf.setSslContext(TransportSupport.createJdkSslContext(to));
        // 构造用户名
        String userName = "accessKey=" + huaWeiIotConfigProperties.getAccessKey();
        // 设置扩展属性，添加时间戳
        cf.setExtension(JmsConnectionExtensions.USERNAME_OVERRIDE.toString(), (connection, uri) -> {
            // IoTDA的userName组成格式如下："accessKey=${accessKey}|timestamp=${timestamp}"
            String newUserName = userName;
            // 如果连接是JmsConnection实例，则获取其用户名
            if (connection instanceof JmsConnection) {
                newUserName = ((JmsConnection) connection).getUsername();
            }
            // 返回带时间戳的用户名
            return newUserName + "|timestamp=" + System.currentTimeMillis();
        });

        // 创建连接
        return cf.createConnection(userName, huaWeiIotConfigProperties.getAccessCode());
    }

    /**
     * 生成amqp连接地址
     *
     * @return amqp连接地址
     */
    public String generateConnectUrl() {
        // 构造基础URI，根据是否使用SSL选择协议
        String uri = MessageFormat.format("{0}://{1}:{2}",
                (huaWeiIotConfigProperties.isUseSsl() ? "amqps" : "amqp"),
                huaWeiIotConfigProperties.getHost(),
                String.valueOf(huaWeiIotConfigProperties.getPort()));
        // 创建URI选项映射
        Map<String, String> uriOptions = new HashMap<>();
        // 设置虚拟主机
        uriOptions.put("amqp.vhost", huaWeiIotConfigProperties.getVhost());
        // 设置空闲超时时间
        uriOptions.put("amqp.idleTimeout", String.valueOf(huaWeiIotConfigProperties.getIdleTimeout()));
        // 设置SASL机制
        uriOptions.put("amqp.saslMechanisms", huaWeiIotConfigProperties.getSaslMechanisms());

        // 创建JMS选项映射
        Map<String, String> jmsOptions = new HashMap<>();
        // 设置队列预取策略
        jmsOptions.put("jms.prefetchPolicy.queuePrefetch", String.valueOf(huaWeiIotConfigProperties.getQueuePrefetch()));
        // 设置客户端ID
        if (CharSequenceUtil.isNotBlank(clientId)) {
            jmsOptions.put("jms.clientID", clientId);
        } else {
            // 如果clientId为空，则使用UUID
            jmsOptions.put("jms.clientID", UUID.randomUUID().toString());
        }
        // 设置重连延迟
        jmsOptions.put("failover.reconnectDelay", String.valueOf(huaWeiIotConfigProperties.getReconnectDelay()));
        // 设置最大重连延迟
        jmsOptions.put("failover.maxReconnectDelay", String.valueOf(huaWeiIotConfigProperties.getMaxReconnectDelay()));
        // 如果最大重试次数大于0，则设置最大重试次数
        if (huaWeiIotConfigProperties.getMaxReconnectAttempts() > 0) {
            jmsOptions.put("failover.maxReconnectAttempts", String.valueOf(huaWeiIotConfigProperties.getMaxReconnectAttempts()));
        }
        // 处理扩展选项
        if (huaWeiIotConfigProperties.getExtendedOptions() != null) {
            // 遍历扩展选项
            for (Map.Entry<String, String> option : huaWeiIotConfigProperties.getExtendedOptions().entrySet()) {
                // 根据选项键的前缀决定放入哪个映射中
                if (option.getKey().startsWith("amqp.") || option.getKey().startsWith("transport.")) {
                    uriOptions.put(option.getKey(), option.getValue());
                } else {
                    jmsOptions.put(option.getKey(), option.getValue());
                }
            }
        }
        // 构造最终的连接URL
        StringBuilder stringBuilder = new StringBuilder();
        // 添加URI选项
        stringBuilder.append(uriOptions.entrySet().stream()
                .map(option -> MessageFormat.format("{0}={1}", option.getKey(), option.getValue()))
                .collect(Collectors.joining("&", "failover:(" + uri + "?", ")")));
        // 添加JMS选项
        stringBuilder.append(jmsOptions.entrySet().stream()
                .map(option -> MessageFormat.format("{0}={1}", option.getKey(), option.getValue()))
                .collect(Collectors.joining("&", "?", "")));
        // 返回构造好的连接URL
        return stringBuilder.toString();
    }

    /**
     * 创建消费者
     *
     * @param session    session
     * @param connection amqp连接
     * @param queueName  队列名称
     * @return 消费者
     */
    public MessageConsumer newConsumer(Session session, Connection connection, String queueName) throws Exception {
        // 检查连接是否有效
        if (connection == null || !(connection instanceof JmsConnection) || ((JmsConnection) connection).isClosed()) {
            // 连接无效时抛出异常
            throw new Exception("create consumer failed,the connection is disconnected.");
        }

        // 创建并返回队列消费者
        return session.createConsumer(new JmsQueue(queueName));
    }

    // 消息监听器，用于处理接收到的消息
    private final MessageListener messageListener = message -> {
        try {
            // 异步处理收到的消息，确保onMessage函数里没有耗时逻辑
            threadPoolTaskExecutor.submit(() -> processMessage(message));
        } catch (Exception e) {
            // 记录提交任务时的异常
            log.error("submit task occurs exception ", e);
        }
    };


    @Autowired
    private IDeviceDataService deviceDataService;

    /**
     * 在这里处理您收到消息后的具体业务逻辑。
     * 
     * @param message 接收到的消息
     */
    private void processMessage(Message message) {
        // 消息内容字符串
        String contentStr;
        try {
            // 获取消息体内容
            contentStr = message.getBody(String.class);
            // 获取消息主题
            String topic = message.getStringProperty("topic");
            // 获取消息ID
            String messageId = message.getStringProperty("messageId");
            // 记录接收到的消息信息
            log.info("receive message,\n topic = {},\n messageId = {},\n content = {}", topic, messageId, contentStr);
        } catch (JMSException e) {
            // JMS异常时抛出运行时异常
            throw new RuntimeException("服务器错误");
        }

        //将消息转换为json格式，数据为空，直接返回
        JSONObject jsonMsg = JSONUtil.parseObj(contentStr);

        //取出消息中的notify_data字段，数据为空，直接返回
        JSONObject jsonNotifyData = jsonMsg.getJSONObject("notify_data");
        if (ObjectUtil.isEmpty(jsonMsg)) {
            return;
        }

        //json格式转换为对象，如果属性上报为空，程序结束
        IotMsgNotifyData iotMsgNotifyData = JSONUtil.toBean(jsonNotifyData, IotMsgNotifyData.class);
        if (ObjectUtil.isEmpty(iotMsgNotifyData.getBody()) || ObjectUtil.isEmpty(iotMsgNotifyData.getBody().getServices())) {
            return;
        }

        deviceDataService.batchInsertDeviceData(iotMsgNotifyData);

    }

    // JMS连接监听器，用于监听连接状态变化
    private final JmsConnectionListener myJmsConnectionListener = new JmsConnectionListener() {
        /**
         * 连接成功建立。
         * 
         * @param remoteURI 远程URI
         */
        @Override
        public void onConnectionEstablished(URI remoteURI) {
            // 记录连接建立日志
            log.info("onConnectionEstablished, remoteUri:{}", remoteURI);
        }

        /**
         * 尝试过最大重试次数之后，最终连接失败。
         * 
         * @param error 错误信息
         */
        @Override
        public void onConnectionFailure(Throwable error) {
            // 记录连接失败日志
            log.error("onConnectionFailure, {}", error.getMessage());
        }

        /**
         * 连接中断。
         * 
         * @param remoteURI 远程URI
         */
        @Override
        public void onConnectionInterrupted(URI remoteURI) {
            // 记录连接中断日志
            log.info("onConnectionInterrupted, remoteUri:{}", remoteURI);
        }

        /**
         * 连接中断后又自动重连上。
         * 
         * @param remoteURI 远程URI
         */
        @Override
        public void onConnectionRestored(URI remoteURI) {
            // 记录连接恢复日志
            log.info("onConnectionRestored, remoteUri:{}", remoteURI);
        }

        /**
         * 接收到入站消息
         * 
         * @param envelope 消息封装对象
         */
        @Override
        public void onInboundMessage(JmsInboundMessageDispatch envelope) {
            // 空实现
        }

        /**
         * 会话关闭
         * 
         * @param session 会话对象
         * @param cause   关闭原因
         */
        @Override
        public void onSessionClosed(Session session, Throwable cause) {
            // 空实现
        }

        /**
         * 消费者关闭
         * 
         * @param consumer 消费者对象
         * @param cause    关闭原因
         */
        @Override
        public void onConsumerClosed(MessageConsumer consumer, Throwable cause) {
            // 空实现
        }

        /**
         * 生产者关闭
         * 
         * @param producer 生产者对象
         * @param cause    关闭原因
         */
        @Override
        public void onProducerClosed(MessageProducer producer, Throwable cause) {
            // 空实现
        }
    };
}